package trabajo.sw;

import java.io.Serializable;
import java.util.List;
import javax.faces.model.ListDataModel;
import clases.domain.Pieza;
import org.primefaces.model.SelectableDataModel;

public class PiezaDataModal extends ListDataModel<Pieza> implements SelectableDataModel<Pieza>, Serializable {  

    public PiezaDataModal() {
    }

    public PiezaDataModal(List<Pieza> data) {
        super(data);
    }
   
    @Override
    public Pieza getRowData(String rowKey) {
       
        List<Pieza> piezas = (List<Pieza>) getWrappedData();
       
        for(Pieza pieza : piezas) {
            if(String.valueOf(pieza.getNumPieza()).equals(rowKey.trim()))
                return pieza;
        }
       
        return null;
    }

    @Override
    public Object getRowKey(Pieza pieza) {
        return pieza.getNumPieza();
    }
}
